#pragma once
#include <iostream>
#include <string>
#include <sys/socket.h>
#include <sys/types.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include<cstring>
// 网络地址和主机地址之间进行转换的类

class InetAddr
{
public:
    // 拿网络地址来构造，内部实现网络地址转主机地址
    InetAddr(struct sockaddr_in &addr) : _addr(addr)
    {
        _port = ntohs(_addr.sin_port);           // 从网络中拿到的！网络序列
        //_ip = inet_ntoa(_addr.sin_addr); // 4字节网络风格的IP -> 点分十进制的字符串风格的IP。inet_ntoa用到了静态空间，存在线程安全的问题。
        char ipbuffer[64];
        inet_ntop(AF_INET,&_addr.sin_addr,ipbuffer,sizeof(ipbuffer)); // 4字节网络风格的IP -> 点分十进制的字符串风格的IP。自己传入了一个缓冲区，内部转化时把转化结果放在用户缓冲区里，所以这个函数是线程安全的
        _ip=ipbuffer;
    }

    // 拿主机地址来构造，内部实现主机地址转网络地址
    InetAddr(const std::string &ip,uint16_t port)
        :_ip(ip),_port(port)
    {
        memset(&_addr,0,sizeof(_addr));
        _addr.sin_family=AF_INET;
        //_addr.sin_addr.s_addr=inet_addr(_ip.c_str()); // 功能：1、IP转成4字节  2、4字节转成网络序列
        inet_pton(AF_INET,_ip.c_str(),&_addr.sin_addr); // 同上
        _addr.sin_port=htons(_port);
    }

    uint16_t Port() {return _port;}
    std::string Ip() {return _ip;}
    const struct sockaddr_in &NetAddr() { return _addr; }
    bool operator==(const InetAddr &addr)
    {
        return addr._ip == _ip && addr._port == _port;
    }
    std::string StringAddr()
    {
        return _ip + ":" + std::to_string(_port);
    }
    ~InetAddr()
    {}

private:
    struct sockaddr_in _addr;
    std::string _ip;
    uint16_t _port;
};
